Solve differential equations in Python